import matplotlib.pyplot as plt

plt.rcParams["font.sans-serif"] = ["SimHei"]  # 设置字体
plt.rcParams["axes.unicode_minus"] = False  # 正常显示负号
import pandas as pd


def dataFind(path, flag):
    x = []
    v1 = []
    v2 = []
    v3 = []
    v4 = []
    v5 = []
    v6 = []
    v7 = []
    v8 = []
    data = pd.read_excel(path)
    for i in range(0, len(data['ASM']) - 1):
        x.append(i)
        v1.append(data['avem'][i])
        v2.append(data['aved'][i])
        v3.append(data['avenum'][i])
        v4.append(data['dissimilarity'][i])
        v5.append(data['homogeneity'][i])
        v6.append(data['ASM'][i])
        v7.append(data['energy'][i])
        v8.append(data['correlation'][i])

    font1 = {'weight': 'normal', 'size': 20}

    if flag == 0:
        N = 3
        M = 2
        # 形成NxM大小的画布

        plt.figure(figsize=(32, 16))
        plt.figure(1)
        plt.subplot(N, M, 1)
        # plt.figure(figsize=(16, 8))
        plt.title('絮体平均面积大小关系', fontsize=10, pad=15)
        plt.rcParams["font.sans-serif"] = ["SimHei"]  # 设置字体
        plt.rcParams["axes.unicode_minus"] = False  # 正常显示负号
        # plt.xlabel('图像序号', font1)
        # plt.ylabel('絮体平均面积', font1)
        plt.grid(True)
        plt.plot(x, v1)
        plt.xticks([])
        plt.yticks([])

        plt.subplot(N, M, 2)
        # plt.figure(figsize=(16, 8))
        plt.title('絮体数量个数关系', fontsize=10, pad=15)
        # plt.xlabel('picture num', font1)
        # plt.ylabel('average count', font1)
        plt.grid(True)
        plt.plot(x, v3)
        plt.xticks([])
        plt.yticks([])
        # plt.figure(figsize=(16, 8))

        plt.subplot(N, M, 5)
        plt.title('纹理对比度关系', fontsize=10, pad=15)
        # plt.xlabel('picture num', font1)
        # plt.ylabel('average dissimilarity', font1)
        plt.grid(True)
        plt.plot(x, v4)
        plt.xticks([])
        plt.yticks([])
        # plt.show()
        #
        # plt.figure(figsize=(16, 8))
        plt.subplot(N, M, 3)
        # plt.title('average homogeneity relation', fontsize=10, pad=15)
        plt.title('纹理逆差矩关系', fontsize=10, pad=15)
        # plt.xlabel('picture num', font1)
        # plt.ylabel('average homogeneity', font1)
        plt.grid(True)
        plt.plot(x, v5)
        plt.xticks([])
        plt.yticks([])
        # plt.show()
        #
        # plt.figure(figsize=(16, 8))
        plt.subplot(N, M, 4)
        plt.title('纹理角二阶矩关系', fontsize=10, pad=15)
        # plt.xlabel('picture num', font1)
        # plt.ylabel('average ASM', font1)
        plt.grid(True)
        plt.plot(x, v6)
        plt.xticks([])
        plt.yticks([])
        # plt.show()
        #
        # plt.figure(figsize=(16, 8))

        # plt.show()
        #
        # plt.figure(figsize=(16, 8))
        plt.subplot(N, M, 6)
        plt.title('纹理相关性关系', fontsize=10, pad=15)
        # plt.xlabel('picture num', font1)
        # plt.ylabel('average correlation', font1)
        plt.grid(True)
        plt.plot(x, v8)
        plt.xticks([])
        plt.yticks([])
        plt.show()
    elif flag == 1:
        plt.figure(figsize=(16, 8))
        plt.title('平均面积关系', fontsize=20, pad=30)
        plt.rcParams["font.sans-serif"] = ["SimHei"]  # 设置字体
        plt.rcParams["axes.unicode_minus"] = False  # 正常显示负号
        plt.xlabel('图像序号', font1)
        plt.ylabel('图像中絮体平均面积', font1)
        plt.grid(True)
        plt.plot(x, v1)
        plt.show()

    elif flag == 2:
        plt.figure(figsize=(16, 8))
        plt.title('絮体个数关系', fontsize=20, pad=30)
        plt.rcParams["font.sans-serif"] = ["SimHei"]  # 设置字体
        plt.rcParams["axes.unicode_minus"] = False  # 正常显示负号
        plt.xlabel('图像序号', font1)
        plt.ylabel('图像中絮体个数', font1)
        plt.grid(True)
        plt.plot(x, v3)
        plt.show()
    elif flag == 5:
        plt.figure(figsize=(16, 8))
        plt.title('纹理对比度关系', fontsize=20, pad=30)
        plt.rcParams["font.sans-serif"] = ["SimHei"]  # 设置字体
        plt.rcParams["axes.unicode_minus"] = False  # 正常显示负号
        plt.xlabel('图像序号', font1)
        plt.ylabel('图像中纹理对比度', font1)
        plt.grid(True)
        plt.plot(x, v4)
        plt.show()
    elif flag == 3:
        plt.figure(figsize=(16, 8))
        plt.title('纹理逆差矩关系', fontsize=20, pad=30)
        plt.rcParams["font.sans-serif"] = ["SimHei"]  # 设置字体
        plt.rcParams["axes.unicode_minus"] = False  # 正常显示负号
        plt.xlabel('图像序号', font1)
        plt.ylabel('图像中纹理逆差矩', font1)
        plt.grid(True)
        plt.plot(x, v5)
        plt.show()
    elif flag == 4:
        plt.figure(figsize=(16, 8))
        plt.title('角二阶矩关系', fontsize=20, pad=30)
        plt.rcParams["font.sans-serif"] = ["SimHei"]  # 设置字体
        plt.rcParams["axes.unicode_minus"] = False  # 正常显示负号
        plt.xlabel('图像序号', font1)
        plt.ylabel('图像中纹理角二阶矩', font1)
        plt.grid(True)
        plt.plot(x, v6)
        plt.show()

    elif flag == 6:
        plt.figure(figsize=(16, 8))
        plt.title('相关性关系', fontsize=20, pad=30)
        plt.rcParams["font.sans-serif"] = ["SimHei"]  # 设置字体
        plt.rcParams["axes.unicode_minus"] = False  # 正常显示负号
        plt.xlabel('图像序号', font1)
        plt.ylabel('图像中纹理相关性', font1)
        plt.grid(True)
        plt.plot(x, v8)
        plt.show()
